diff --git a/recovery/Android.bp b/recovery/Android.bp
index 17dde5f..0fd8906 100644
--- a/recovery/Android.bp
+++ b/recovery/Android.bp
@@ -26,7 +26,8 @@
     srcs: [
         "recovery_ui.cpp",
     ],
-    include_dirs: [
-        "bootable/recovery"
+
+    shared_libs: [
+        "librecovery_ui",
     ],
 }
diff --git a/recovery/recovery_ui.cpp b/recovery/recovery_ui.cpp
index d67ff5f..e82be6a 100644
--- a/recovery/recovery_ui.cpp
+++ b/recovery/recovery_ui.cpp
@@ -14,8 +14,8 @@
  * limitations under the License.
  */
 
-#include <device.h>
-#include <screen_ui.h>
+#include <recovery_ui/device.h>
+#include <recovery_ui/screen_ui.h>
 
 namespace android {
 namespace device {